import { NodeModel, NodeModelGenerics, PortModelAlignment } from '@projectstorm/react-diagrams';
import { DatabasePortModel } from './DatabasePortModel';
import { DATABASE_NODE_MODEL_TYPE } from './constants';

export interface DatabaseNodeModelGenerics {
    PORT: DatabasePortModel;
}

export class DatabaseNodeModel extends NodeModel<NodeModelGenerics & DatabaseNodeModelGenerics> {
    constructor() {
        super({ type: DATABASE_NODE_MODEL_TYPE });
        this.addPort(new DatabasePortModel(PortModelAlignment.TOP));
    }
}